# 连接云服务器 MongoDB（需替换为你的实际信息）
from pymongo import MongoClient
mongo_client = MongoClient(
    host="47.111.13.97",
    port=27017,  # 改为整数
    username="admin",
    password="123456",
    authSource="admin"
)

# 选择数据库（若不存在会自动创建）
mongo_db = mongo_client["admin"]  # 数据库名可自定义

# ---------------------- 创建评价表（comment） ----------------------
comment_collection = mongo_db["comment"]  # 集合（表）名

# 创建唯一索引（确保 id 唯一）
comment_collection.create_index("id", unique=True)

# 插入一条评论示例数据
comment_collection.insert_one({
    "id": "c1001",               # 评论 ID（自定义，需唯一）
    "user_id": "u1001",          # 用户 ID（后续可替换为真实用户或水军账号）
    "movie_id": "35660795",      # 电影 ID（对应你要爬取的电影）
    "content": "这是一条测试评论",
    "type": "好评",              # 类型：好评/中评/差评
    "time": "2025-06-09 14:30:00"  # 时间格式建议用 ISO 8601（如 "2025-06-09T14:30:00Z"）
})

# ---------------------- 创建用户表（user）并插入水军账号 ----------------------
user_collection = mongo_db["user"]  # 集合（表）名

# 插入两个水军账号
user_collection.insert_many([
    {
        "user_id": "fake_user_001",
        "username": "水军一号",
        "is_bot": True  # 标记为水军（True/False）
    },
    {
        "user_id": "fake_user_002",
        "username": "水军二号",
        "is_bot": True
    }
])

# 关闭连接（可选，程序结束时会自动关闭）
mongo_client.close()

print("表结构和测试数据已创建完成！")

# ...（原有代码）

if __name__ == '__main__':
    # 原有插入操作会执行，之后添加查询
    # 查询评价表的示例数据
    comment_data = comment_collection.find_one({"id": "c1001"})
    print("查询到的评价数据：", comment_data)

    # 查询用户表的水军数据
    user_data = list(user_collection.find({"is_bot": True}))
    print("查询到的水军用户：", user_data)